package ai.zixing.mashibing.new_class.class05;

import java.util.HashSet;

// 位图
public class Code02_BitMap2 {

    public static class BitMap {
        private long[] bits;

        public BitMap(int max) {
            // long 占用 8 个字节，64个bit
            // (max + 64) / 64 可以计算需要占用的数组大小
            bits = new long[(max + 64) >> 6];
        }

        public void add(int num) {
            // num % 64 就是 num & 63 【num & 63效率快】
            // 63 就是 64 对应的 1 的位置全是 1 的
            // 需要将num 对应的位置改为 1
            // 【【【1L 是 64位， 1是32 位 需要注意，位运算可能出错】】】
            bits[num >> 6] |= (1L << (num & 63));
        }

        public void delete(int num) {
            // 将需要删除的位置置为 0，与运算后即删除
            // 和add方法存入时取反则即时需要删除的位置为 0
            bits[num >> 6] &= ~(1L << (num & 63));
        }

        public boolean contains(int num) {
            return (bits[num >> 6] & (1L << (num & 63))) != 0;
        }
    }

    public static void main(String[] args) {
        System.out.println("begin test");
        int max = 10000;
        BitMap bitMap = new BitMap(max);
        HashSet<Integer> set = new HashSet<>();
        int testTime = 10000000;
        for (int i = 0; i < testTime; i++) {
            int num = (int) (Math.random() * max);
            double decide = Math.random();
            if (decide < 0.33) {
                bitMap.add(num);
                set.add(num);
            } else if (decide < 0.66) {
                bitMap.delete(num);
                set.remove(num);
            } else {
                if (bitMap.contains(num) != set.contains(num)) {
                    System.out.println("Oops!");
                    break;
                }
            }
        }
        for (int i = 0; i <= max; i++) {
            if (bitMap.contains(i) != set.contains(i)) {
                System.out.println("Oops!");
            }
        }
        System.out.println("end test");
    }

}
